package main

import (
	"fmt"

	"github.com/spf13/viper"
)

func initConfig() error {
	viper.SetConfigName("config")
	viper.SetConfigType("yaml")
	viper.AddConfigPath(".")

	if err := viper.ReadInConfig(); err != nil {
		if _, ok := err.(viper.ConfigFileNotFoundError); ok {
			fmt.Println("配置文件未找到")
		} else {
			fmt.Printf("读取配置文件出错: %v\n", err)
		}
		return err
	}
	return nil
}

func getDB() {
	host := viper.GetString("database.host")
	port := viper.GetInt("database.port")
	username := viper.GetString("database.username")
	password := viper.GetString("database.password")
	dbname := viper.GetString("database.dbname")

	fmt.Println(host, port, username, password, dbname)

}

func main() {
	if err := initConfig(); err != nil {
		return
	}

	getDB()
	// 连接数据库
	fmt.Println("成功连接到数据库")
}
